package com.example.demo.service.weCon.impl;

import com.example.demo.config.WeConConfig.Media;
import com.example.demo.entity.DTO.MediaUploadDTO;
import com.example.demo.enums.MediaTypeEnum;
import com.example.demo.service.weCon.UploadService;
import com.example.demo.service.weCon.WeConService;
import com.example.demo.utils.GsonUtils;
import com.example.demo.utils.WeChatHttpClient;
import java.io.ByteArrayOutputStream;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.HashMap;
import java.util.Map;
import javax.servlet.http.HttpServletResponse;
import javax.validation.constraints.NotNull;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

/**
 * @author jack
 * @data 2021/8/16 4:12 下午
 */
@Service
public class UploadServiceImpl implements UploadService {

  @Autowired
  private WeConService weConService;


  @Override
  public MediaUploadDTO upload(@NotNull MultipartFile file, @NotNull MediaTypeEnum mediaTypeEnum) {
    Map<String, Object> paramMap = new HashMap<>();
    paramMap.put("type", mediaTypeEnum.getCode());
    String result = WeChatHttpClient.doPostUpload(
        weConService.generateWeConUrl(Media.UPLOAD, this.getWeConFunctionType(), paramMap), file);
    return GsonUtils.fromJson(result, MediaUploadDTO.class);
  }

  @Override
  public void get(HttpServletResponse response, String mediaId) throws Exception {
    String url = weConService.generateWeConUrl(Media.GET, this.getWeConFunctionType());
    Map<String, String> paramMap = new HashMap<>();
    paramMap.put("media_id", mediaId);
    InputStream inputStream = WeChatHttpClient.doGetInputStreamByMap(url, paramMap);
    //得到图片的二进制数据，以二进制封装得到数据，具有通用性
    byte[] data = readInputStream(inputStream);
    //new一个文件对象用来保存图片，默认保存当前工程根目录
    //File imageFile = new File(filename);
    //创建输出流
    OutputStream outStream = response.getOutputStream();
    //写入数据
    outStream.write(data);
    //关闭输出流
    outStream.close();
  }

  @Override
  public String uploadimg(MultipartFile file) {
    String url = weConService.generateWeConUrl(Media.uploadimg, this.getWeConFunctionType());
    String jsonStr = WeChatHttpClient.doPostUpload(url, file);
    return WeChatHttpClient.filterResultReturnOne(jsonStr);
  }

  /**
   * 得到图片的二进制数据，以二进制封装得到数据，具有通用性
   *
   * @param inStream
   * @return
   * @throws Exception
   */
  public static byte[] readInputStream(InputStream inStream) throws Exception {
    ByteArrayOutputStream outStream = new ByteArrayOutputStream();
    //创建一个Buffer字符串
    byte[] buffer = new byte[1024];
    //每次读取的字符串长度，如果为-1，代表全部读取完毕
    int len = 0;
    //使用一个输入流从buffer里把数据读取出来
    while ((len = inStream.read(buffer)) != -1) {
      //用输出流往buffer里写入数据，中间参数代表从哪个位置开始读，len代表读取的长度
      outStream.write(buffer, 0, len);
    }
    //关闭输入流
    inStream.close();
    //把outStream里的数据写入内存
    return outStream.toByteArray();
  }
}
